import torch
from torch import nn
print(torch.cuda.is_available())


class MyNetWork(nn.Module):
    def __init__(self):
        super.__init__(MyNetWork,self)
        self.net = nn.Sequential(
            nn.Linear(3, 40),
            nn.Sigmoid(),
            nn.Linear(40, 1)
        )

    def forward(self,x):
        return self.net()

layer=nn.Linear(32,64)
print(layer)

